import random
import json
import os

def generate_sorted_array(size=101, min_val=0, max_val=100):
    """生成0-100的完全有序数组"""
    return list(range(min_val, max_val + 1))

def generate_random_array(size=101, min_val=0, max_val=100):
    """生成0-100的随机整数数组"""
    # 先生成有序数组再打乱
    arr = list(range(min_val, max_val + 1))
    random.shuffle(arr)
    return arr

def generate_reversed_array(size=101):
    """生成0-100的逆序数组"""
    return list(range(100, -1, -1))

if __name__ == "__main__":
    # 创建data目录（如果不存在）
    os.makedirs("data", exist_ok=True)
    
    # 生成测试数据
    test_data = {
        "sorted": generate_sorted_array(),
        "random": generate_random_array(),
        "reversed": generate_reversed_array()
    }
    
    # 保存为JSON文件
    with open("data/test_data.json", "w") as f:
        json.dump(test_data, f, indent=4)